from problems.TREE import Tree


class TREE2(Tree):
    def __init__(self, var_dim: int, obj_dim: int, max_fun_eval, kwargs: dict):
        super().__init__(var_dim, obj_dim, max_fun_eval, 1, 3, self.init_dec_boundary0, 2, kwargs)

    def eval_value(self, pop_dec):
        return super().eval_value_mix(pop_dec, self.calc_ea0)
